
from ._extended_torch_module import ExtendedTorchModule